ForeignKeyNamingStrategy.java

package org.codefilarete.stalactite.dsl.naming;

import java.math.BigInteger;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Comparator;

import org.codefilarete.stalactite.query.model.JoinLink;
import org.codefilarete.stalactite.sql.ddl.structure.Column;
import org.codefilarete.stalactite.sql.ddl.structure.Key;
import org.codefilarete.stalactite.sql.ddl.structure.Table;
import org.codefilarete.tool.exception.Exceptions;
import org.codefilarete.tool.trace.MutableInt;

import static org.codefilarete.tool.collection.Iterables.first;

/**
 * Contract for giving a name to a foreign key
 *
 * @author Guillaume Mary
 */
public interface ForeignKeyNamingStrategy {
	
	String DEFAULT_FOREIGNKEY_PREFIX = "FK_";
	
	Comparator<JoinLink<?, ?>> COLUMN_COMPARATOR = Comparator.comparing(JoinLink::getExpression);
	
	<SOURCETABLE extends Table<SOURCETABLE>, TARGETTABLE extends Table<TARGETTABLE>, ID> String giveName(Key<SOURCETABLE, ID> src, Key<TARGETTABLE, ID> target);
	
	ForeignKeyNamingStrategy HASH = new ForeignKeyNamingStrategy() {
		@Override
		public <SOURCETABLE extends Table<SOURCETABLE>, TARGETTABLE extends Table<TARGETTABLE>, ID> String giveName(Key<SOURCETABLE, ID> src, Key<TARGETTABLE, ID> target) {
			MutableInt hashCode = new MutableInt(src.getTable().getAbsoluteName().hashCode());
			hashCode.reset(hashCode.getValue() * 31 + target.getTable().getAbsoluteName().hashCode());
			// We ensure a consistent ordering of columns, regardless of the order they were bound.
			src.getColumns().stream().sorted(COLUMN_COMPARATOR).forEach(joinLink -> {
				hashCode.reset(hashCode.getValue() * 31 + joinLink.getExpression().hashCode());
			});
			target.getColumns().stream().sorted(COLUMN_COMPARATOR).forEach(joinLink -> {
				hashCode.reset(hashCode.getValue() * 31 + joinLink.getExpression().hashCode());
			});
			return DEFAULT_FOREIGNKEY_PREFIX + Integer.toHexString(hashCode.getValue());
		}
	};
	
	/**
	 * Composed of a prefix, source table name, source column name, target column name.
	 * Must be quite unique to prevent Database such as HSQLDB to yel because key names uniqueness is over a table space, not per table.
	 */
	ForeignKeyNamingStrategy DEFAULT = new ForeignKeyNamingStrategy() {
		@Override
		public <SOURCETABLE extends Table<SOURCETABLE>, TARGETTABLE extends Table<TARGETTABLE>, ID> String giveName(Key<SOURCETABLE, ID> src, Key<TARGETTABLE, ID> target) {
			if (src.getColumns().size() == 1) {
				return DEFAULT_FOREIGNKEY_PREFIX + src.getTable().getName() + "_" + ((Column) first(src.getColumns())).getName()
						+ "_" + target.getTable().getName() + "_" + ((Column) first(target.getColumns())).getName();
			} else {
				return HASH.giveName(src, target);
			}
		}
	};
	
	/**
	 * Generates same name as Hibernate (4.3.7) does. From org.hibernate.mapping.Constraint#generateName(String, Table, Column... columns)
	 */
	ForeignKeyNamingStrategy HIBERNATE_4 = new ForeignKeyNamingStrategy() {
		@Override
		public <SOURCETABLE extends Table<SOURCETABLE>, TARGETTABLE extends Table<TARGETTABLE>, ID> String giveName(Key<SOURCETABLE, ID> src, Key<TARGETTABLE, ID> target) {
			// Use a concatenation that guarantees uniqueness, even if identical names
			// exist between all table and column identifiers.
			StringBuilder sb = new StringBuilder("table`" + src.getTable().getName() + "`");
			
			// We ensure a consistent ordering of columns, regardless of the order they were bound.
			target.getColumns().stream().sorted(COLUMN_COMPARATOR).forEach(column -> sb.append("column`").append(column.getExpression()).append("`"));
			return "FK" + hashName(sb.toString());
		}
	};
	
	/**
	 * Generates a hash for a foreign key represented by given arguments
	 * Implementation that returns the same hash as Hibernate 7 does.
	 * From org.hibernate.boot.model.naming.NamingHelper#generateHashedFkName(String, Identifier, Identifier, List&lt;Identifier&gt;...)
	 */
	ForeignKeyNamingStrategy HIBERNATE_7 = new ForeignKeyNamingStrategy() {
		@Override
		public <SOURCETABLE extends Table<SOURCETABLE>, TARGETTABLE extends Table<TARGETTABLE>, ID> String giveName(Key<SOURCETABLE, ID> src, Key<TARGETTABLE, ID> target) {
			// Use a concatenation that guarantees uniqueness, even if identical names
			// exist between all table and column identifiers.
			StringBuilder sb = new StringBuilder("table`" + src.getTable().getName() + "`")
					.append("references`").append(target.getTable().getName()).append("`");
			
			// Note that referenced columns are not taken into account because the primary key is automatically targeted by database engines (same
			// principle as Hibernate 7).
			// We ensure a consistent ordering of columns, regardless of the order they were bound.
			src.getColumns().stream().sorted(COLUMN_COMPARATOR).forEach(column -> sb.append("column`").append(column.getExpression()).append("`"));
			return "FK" + hashName(sb.toString());
		}
	};
	
	/**
	 * Same algorithm as Hibernate (4.3.7), see org.hibernate.mapping.Constraint#hashedName(String)
	 *
	 * @param s the string to be hashed
	 * @return a hashed (MD5) version of the given String, less than 30 characters
	 */
	static String hashName(String s) {
		try {
			@SuppressWarnings("java:S4790" /* No sensitive code related to MD5 usage : only used to hash foreign key name */)
			MessageDigest md = MessageDigest.getInstance("MD5");
			md.update(s.getBytes());
			byte[] digest = md.digest();
			BigInteger bigInt = new BigInteger(1, digest);
			// By converting to base 35 (full alphanumeric), we guarantee
			// that the length of the name will always be smaller than the 30
			// character identifier restriction enforced by a few dialects.
			return bigInt.toString(35);
		} catch (NoSuchAlgorithmException e) {
			throw Exceptions.asRuntimeException(e);
		}
	}
}